Added tests to lab 2