Add condition on white or black