Added in passing of key