Test omego upgrade directly in Travis

Also a couple of flake8 fixes