More changes to the method test class