Added some functionality to the comments

Added some functionality to the comments