Improve tools/layout-checks.py

Instead of hard-coding first entries, use the section headers directly.

Make the output a bit nicer.

Change-Id: I0a5f947d42ada5f010853de04f3ca651039eeb60
Merge "Improve tools/layout-checks.py"