Adding test case for Assignment 2