[FEATURE] Add set and tests for set