Added keyword highlighting