OK, let's write some unit tests