- Handle both arrow functions and traditional function expressions - Add tests to ensure correctness of transformations