Lines to modify : ```python # Loss function / forward pass targets = np.repeat(targets.reshape(-1,1),values.shape[1],axis=1) loss = self.nn.forward(np.array(states), targets) ```