Adjusting demo script