Added plotting routine