Added more tests to the python file, and changed how certain functions return
Additional unittests