Correcting the last two test cases for extra functions