completing unit tests