Add basic unittest