- Add missing partials.
- Add state classes