Improve tvtk import 

And some PEP8 formatting fixes
use a specific exception here.