Use key in dict instead of key in dict.keys()

Use key in dict instead of key in dict.keys() #317