Update functions tests in main