Add Ben's updated model