Add two methods to test module