Finishing touches for v1 of registration flow

* fixing tests
* getting coverage to 100%
* fixing flake8 issues