Improved model loading and management