Update script

Signed-off-by: benjamincharlton <benjicharlton@gmail.com>