Added update_scenario functionality.
Moved testing block back to the bottom of the program.