Parameterize script to match writeup