[In_progress] Fixed theta in att_viz