Adding a context class to make state sharing easier