Still need to insert shocks code
First cut at complete program