Return module on Python 3