add class for demo