Update experiment so that we can use it in the paper