Fix diagonal line and improve perceiver task script