Add tests to basic func