Added Python 3 support :)