added tests for 2nd task