add last unit tests