Update main.py

Added form for testing