Explicit Unit Return

Added the rule to rewrite asts such that explicit Unit is added to Defn.Def