Cleanup imports

Not, strictly speaking, necessary, but adding the future imports will help with keeping the code python 3 compatible.