Modified testing script to display both plots at the same figure