Add feature to get title