Add actions in functions and Comprehensions