Added tests for for comprehensions