Let's add some unit tests!