Task 2: Create publication-ready and elegant plotting