addedsome unit tests