Slight refactor and show chosen color