first attempt at unit tests