Update main.py

Added lambda on line 35, lets hope that works.