Got to 90% test coverage of python code