Cleaning flakes8 errors