Added tests for assignment 4.