Add functions needed for testing