Add tests that cover basic functionalities