two-stage code snippets in python

simulate launch of a rocket with two stages in python