Made script a bit more useful for testing