implemented done button

better to use toggleClass