change draw function