keenan-burnett Added a light_mode for visualizing final plots. Added more arguments …