Adding models to __init__