Added ability to display the colored solution