Update shocktube.py

Modified code to make it more readable.