Making plot 3 code a bit more elegant