Add logic to determine if done class should be applied