Improve/models

Added __init__ function of models.