Added an option to change the background color