Add state to draw button