Fixed some stuff in plots