Refactored trait methods AST representation to be included in trait AST

Refactored trait methods AST representation to be included in trait AST #160