Start making bake model work, ran into error with associations