Add load_subset_data function and so on. Indent the source code.