Add state changes on track water buttons