simplified grax_context into grax; 50% done on aa tree