Add gradient to output and modify SolveStorage to output only interp