Add API to define final state