Moved plotting to the python/Grid class file.