Updated the test-file

Now contains accurate test function.