Add tests for all methods in module