Add code to experiment (Work still in progress)

Not finished yet.