Added methods to test units
Decided to test all the units in one method