Add check to see if you want to keep going