Fixed small plotting bugs