Added some mock classes for testing. Added a few new tests.
Updated docs