Render image in colab

[jlin27]

In the Neural Network colab, the first image in the tutorial doesn't pull in. It only shows the rst:

It should look like this:

[anikethc]

Hi @jlin27,
I would like to work on this.

[anikethc]

I've observed few more errors in the same Colab file, and I'd be happy to address them.