Fixed pyflakes errors