Simplified the model loader