Working on writing tests for transformation code