Added option to change highlight colour