Add model loading and exceptions