Update plotting script to take a simulation flag as an argument.