tried to make script work with limited models