Revert "Remove color_scheme"

This reverts commit 140e75f5622edac4865a15362291e4610dd0680c.