Unittest

Add unit test