start working on some unit tests (left this a little late)