Personally I'd extract those 2 lines into a method called "set_background_color" and make it a an if-else - but yeah that's probably just me... :-D (nothing bad, just a bit more beautiful to my eyes but I can also do that afterwards)