Error in comments
Improved explanations + inserted code