Update main.py

Added Momentum with Image, to be uploaded.