Added load_map func