Add MOCK_MODE for testing purposes