improve experiment part