changed to conform to new list comprehension syntax

old syntax was used changes were conversion to new one
same as previous commit

same as previous commit
Merge pull request #11 from NikolaMandic/master

old syntax was changed for list comprehension so that library can be used in the current version