second attempt at unit testing