Changed a little bit of code to make the output look more user friendly