Adding click on done