Added code for computing traces
Updated
Plotting traces