Fixed the styles and added some comments for this module