Redid 1 in python; So beautiful