Added some functionality for testing