PSClassMock enhancements