code fix
Added AST function def and call tests