now allowing custom key