Finally fixing the done button