Adding some data setup code