Update for dark theme