modifying experiment code