Starting off with some unit tests...