added some functionality for easier testing