Added return sum and episode G to agent
ReturnSum now updates properly
Added return updating