Updated display and menu code.