changed up some of your code and made some comments