def read_from_folder(folder_name): input_names = [] polynomials_list = [] for root, dirs, files in os.walk(folder_name): input_names.extend(files) for filename in input_names: if filename.endswith('.in'): p = Polynomial() p.read_from_file(folder_name, filename) polynomials_list.append(p) return polynomials_list