Minor update to layout