ENH: Added more code for forward computations