Add ast module