Add basic unit tests