Added code highlighting and some minor styles