cleaned up save.py a bit
wrote load method
added small unit test