Fixed the module loading code to work on both python 2 and python 3.
Might not be the best way to do it, so possibly a hack.