[TestExperiments] Cleaned up the code