Add watch method to fire off a callback on state change