Almost lost my work
Add more comments to describe code