Update simulation.py to match input controlers