Added class done button and function