Minor optimization

Added a comment to my part of the code and removed an unnecessary
variable assignment