Fixed a stupid mistake. Also updated comments.