Add comments to format_drive