Updated function return key