fixed formatting for list comprehensions
added support for generators