My solution to their test