minor update
Updated script with more things