added code to main.py
Maria_LDA pushed (should not have changed anything)