Added extra mock functions for Testing