added a unit tetst