Replace Document AST with Element
Fix import in utils
Work on linked-list style elements