Strongly Typed AST: add unit test.
Small refactor

 - renaming
 - remove useless import