tooth staining

I’m really scared my teeth are gonna be stained when I get my braces off in a couple months. and sometimes I forget to brush my teeth before bed and I know that’s bad. But i’m looking for a good toothpaste that will fight against staining!!