minor changes as I adjust the plotting