When a colleague cries at work...

A coworker passed away last weekend and one of my colleagues was reflecting on the loss and began to cry. I hugged them and offered comforting words, and since i didn’t want to make them uncomfortable, after asking if they're OK, I just slowly walked back to my desk. The more i thought about it, maybe I should have stayed and offered additional hugs and encouraged them to talk more about their feelings. Am I cold? How would you handle the situation?