Nick Diaz training for 2021 comeback