Added the utility functions for testing