Made some changes to Protractor