Update pt.py

cleaned up imports for pt.py to make code more portable and more efficient