Create states_3_1.py

1st challenge for lesson 3 playtime states exercise