Moved the color choice into the main loop