put experiment code back in