Reformat Google co-lab to install correct packages

[mdbresh]

We're creating this issue for HW4!

We need to reformat our Google co-laboratory so that at the top it installs the necessary packages for running the notebook.

[yamanmy]

Installation of some packages were added to the notebook. Hope it works without doing anything!