Make object keys highlight correctly in js