Change plotting to python, improve graphs a bit.