Import cleanup, thanks pyflakes :)