updated main.py

Signed-off-by: ZiharkTheAssassin <twhip265@gmail.com>