put experiment code in proper place