ENH simplify assignment, PEP8