first test case working - prog4