edited code to show pics