Moved mock classes to main folder