adding models to module function def