Add testing script and reference answers
Add problem 3