Create for.py

to use the for loop