Fixed test classes
Added documentation