Adding helper method to get key