Clean the demo script controller_test_2.py