Add comments for clarity of code