beginning to add unit tests