Add transformation logic