Cleaning up and adding some comments
Adding more comments