Add function tests