more tests
Added solution of problem 3