Update main.py

Added test words.