Add login utils
Flake8 fixes