Initial reward model for multitask agent. Still buggy