Updated the comments
Cleaned up the code and updated the comments