changed some lines of code for comments