Let's start adding some unit tests shall we?