def extract_keys(inp_dict, keys):
    """
    Return dictionary witxh selected keys
    :param inp_dict: dictionary
    :param keys: list or string
    :return: dictionary
    """
    if not isinstance(keys, list):
        keys = [keys]
    return {your_key: inp_dict[your_key] for your_key in keys}