Add parts arg to register_parts