add functionality for test checking