Add tests for layout to prevent errors [closes #16]