change highlight method and remove rainbow js