First pass at unit tests