Display buffer now works for the restart prompt. Needs to be incorporated everywhere else.