Add tests about function