Unit tests

We now have Unit Tests!