added functionaliy for local test set
merge