Update codebase to work with Python 3!