Add new methods to tests