Add basic function test for more coverage