Cleaned up the files, added inputs and ready for part 2