Porting plotting to Python 3 as well