Added my python code (Yu)