easier python code