I added a few comments, looks good!