Changed script so that it plays nice with the example data