Continuation/code optimization during live coding interview

Hello, request to have a feature where a coding answer is provided and when interviewer asks a follow up to correct/improve/optimize existing code, when i click, “Answer” button, it creates on explanation to improve, does not continue from the previous “code” answer. Instead we should get explanation + code on the new approach.