Add tagging logic