Extend shell context with models module