Changed a few things in initialize_parts, still have problems in train_dpm that I need to fix