Add some unit tests Review the code (be a little more pythonic, use decorator) Add some logging for debug purposes