Added Done/Undone functionlity