First foray into unit tests