remove extra plotting code