https://progiez.com/nptel-python-for-data-science-assignment-2-answers