fixed first test and added another