Added docstring to the main program