wrote some unit tests