add attention functioin