Add logic for tracking download progress