Make keys an actual function