added code to track progress