Clean up machine_type.py