Fixed generator warning
Small change
Removed comment
Added dict comprehension
Made small change
Deleted alternatives as suggested by @AmeBel
Merge pull request #944 from sebastianruder/small_fix

Fix for #735