fix some plotting things
fix some plotting things