created a function to update the color. fixed import
flake8
added some comments to the tests
make it executable
added some documentation. writing plots to file
flake8 and stuff
Merge branch 'master' of https://github.com/alexfikl/school-work