Added identity function test