Added updated test and expected output for new API