Fix docs model discovery
Add test to ensure models displayed in docs
Add before/after middleware, remove setup