Two new methods added to main.py