getting close with first draft of unit testing