Add more comments to each function