Add class for selenium test