Remove extra print, flake8 fix