added unit test part.