Initial attempt at unit tests