Added code highlighting & make example more simple