My solution for module 3 shock-tube problem