Update main layout and styling