Added plot4
Refactored plots by using 'with'