Added tests in Part_2