Fix hard-coded colors and improved code to accept stylesheets