adding doc-strings to main.py