Updated extension to change icon colour when activated