Improve style throughout experiment code