Small changes

added comments. edited the outputs.