Edited code

Added a comment