Modify Done button