Cleaned up model loading.