Add methods to set tokens