Update example code

I think we want to remove class on complete instead of adding twice.