Add solution and full test to assignment 02