Edited last code cell with syntax highlighting.