Change coloring scheme and add more user control