Add a keys function