write some unit tests