Modify script

Change-Id: I3b30d969bcb4b69f59bc07cc177b6b83bdd36e71
Signed-off-by: Josue Rivera <prbassplayer@gmail.com>