start to add unit tests