Fixing pyflake warnings