Added test code in project4