Fix style error found by flake8
Fix doctest fail on problem 3 and 13