Add docstring for main function