working through unit tests