Added unit tests
Marked classes with missing unit tests