Fix expression wrapping for update part in loops
Fix for_continue tests