Updated import syntax to python3.

Seems to work, haven't performed any extensice testing though. Not
sure if this works with python2 (ST2).