fix keys method to actually return keys and not just print them