Update main.py

Lesson 3 started