Added code to compare different shocks