First attempt at unit tests