[In_progress] Fixed theta in att_viz
[In_progress] Fixed theta in att_viz