add class
fix bug