flesh out unit tests