minor updates to ai