Small update to layout