Update main.py

oops shouldn't raise strings