Add set/unset state callback APIs