Fix pyflakes problems