Update plot.py

Just a small addition that makes all python plots prettier by adding ggplot-like defaults.