Add html class for selenium tests