Remove use of funext-axiom
Adding more documentation, applying JasonGross suggestions.