Import correct modules if running on Python 3.x