Optimize/model schema

Optimize/model schema