Unit Tests
Implemented some unit tests.