improved registration/snapshot interfaces, collect_data, started train_map