Update main.py

Added more methods