Fixed Works only for one model bug #7