Added test for the first problem