Update test framework for functions