add model as __init__ parameter