Forgot to add the mods to direct_collocation.py.