Fixes a bug and adds a comment message.
Fixes a little bug.