Added code to make it easier to replicate experiments