new comment in python solution 3