Fixed Pyflake warnings