Made a few changes to test