Update key generation steps