Added checkbox to keep plot