Small update to main layout