Add method to set payment state.