Added some comments based on feedback