add a draw function