Add done button and action