add labels mapping in order to test multiclass scenario
add posibility of changing class