Update base_models.py

updated rocket_comp