Improved readability of the code.
Refactored the MockExtensions class.