fix bug in flattener, make extensionality axiom explicit