Adding dark/light theme logic