Added some comments
- Just a comment for the main function