Fixed PEP8 violations caused by eariler commits
Added args testsand fixed PEP