python loops and functions