add get_model.py and required stuff