The code for 4th Question, Assignment 1, can be found in this repository. The name of the notebook is sgupta55.ipynb.
Assignment done by Shruti Gupta (sgupta55)
The code can be imported to google colab and run directly, no other data downloads or specifics are required.