Continue docs
Tidy with flake8
Correct comment