Importing libraries

Please check this guide on how to run assignment files on Colab.