added some comments
changed some code i forgot