Add identity() function