Add a redraw function