Added tests for lab4