Be more generic with MonadState instead of State