Added reward model