For Loop Function Code In Python