Revise experiment code