Assignment-01.ipynb fixed