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