Did some more work on the python module