Slight tweak to layout