Implemented also blocks for loops

I'm only somewhat sure this works, but I feel like I should be less
uncertain in these commit messages