Python Wrap Code To Block