Update State.hs