[Truffle] Refactor the module/class declaration nodes.