Writing Python Code With Output Definition