Adding full flow and hooks to Transformer