Updating comments, forgot to do this.