Fixed code styling

Updated code highlighting