Added Model class shityy