comments

I changed how the code looks and added some comments