Initial code with first two plots