update solution to test3