Fixed whitespace issues reported by flake8 test