Add more documentation for ast package