Added comment to the python function