Modified the algorithm to save callee info on the individual call nodes