Fixes code based on JS review.
Merge branch 'uiupdate' of https://github.com/jupyter/colaboratory into uiupdate