Add color setting function.
Even more reformatting