fixed #6, highlight working