- Fixed imports, now working with Python 2 and 3