Writing Python Code With Output Shaft