Edited test cases. Add expected output test result