Fixed a bit of broken code (remove being called twice), added some endl's for style and applied the addendum to get program to run