Improve tvtk import 

And some PEP8 formatting fixes