modifying model loading