Add draw function