Added comments to match instruction video to main.py