python: solved 011