Update color scheme and general cleanup.