Simplified method for done button, to get rid of if statement.