Add Basic Unit Test