Updated script and readme

Updated script and readme