Fixed yo a few pyflakes