Unit Tests

Addresses #4