edited code, added two extra plots