adds visualize_aam function