first python code improvement