Red - added some test cases