Added a few comments throughout