Add model loading code