added some code on protractor