Now the right class is being assigned on clicking
Done