Add retrieval functionality