flak8 fixes
flak8 stuff