Added simple working test for Assignment4