Story/api update

Logic added to customize test script