Added a simple test for assignment 2.