Removed unnecessary plotting code