HW2 Version2

Version2 added the trace back part.