fomart code

code cleanup
follow flake8