First attempt at basic unit testing