Columns with a Matrix Field

columns.html

{% set inColumn = false %}
{% for block in entry.body %}
	{% switch block.type %}
		{% case 'columnBlock' %}
			{% if inColumn %}
				</div>
			{% else %}
				{% set inColumn = true %}
				<div class="lucky grid">
			{% endif %}
			<div class="{{ block.mobileSize }} {{ block.desktopSize }} column">
		{% case 'headerBlock' %}
			<h{{ block.size }}>{{ block.textContent }}</h{{ block.size }}>
		{% case 'textBlock' %}
			{{ block.textContent }}
		{% case 'imageBlock' %}
			<img class="{{ block.position }}" src="{{ block.imageContent.first().url }}">
	{% endswitch %}
{% endfor %}
{% if inColumn %}
	</div></div>
{% endif %}