Unit Tests

Completed the assignment on unit tests.