I noticed that the TODO part of #TODO is highlighted in yellow when editing Python files using Vim. How can I add custom highlighting? For example, I want the BUG part of #BUG to be highlighted in red.