create unit test too.

just adding the unit test too