Adding ability to have a callback for highlighting an item