My solution from the weekend