Fixing the use of key and putting some comments