minor changes to plotting and input