Changed package name + made a small change to the test file