Add: unit tests