added prints to main.py