Handle the dark theme properly