Add identity() function which returns argument unchanged