about my python code