Get rid of canvas.restore()