fixes / updates to experiment code