Small bugs

Hey, i just edited your code, i removed really small bugs. (just while reading the code)