add class back when login