FILE_TYPES = {
    '.json': JsonLoader,
    '.yml': YamlLoader
}

def load_input_data(input_filename):
    file_ext = os.path.splitext(input_filename)[1]
    if file_ext in FILE_TYPES:
        if file_ext == '.json':
            loaded_data = JsonLoader(input_filename).load()
            return  loaded_data
        elif file_ext == '.yml':
            loaded_data = YamlLoader(input_filename).load()
            return loaded_data
    else:
        raise ValueError('Unsupported file type')