Add clear and draw display code