Improve signature in monad-state.sml