Added update after state changes