Tried to fix code highlighting