implemented dan's chroma code; train-test CV split