Sta Rite 333

My pool heater stays on for 2 minutes then stops for 1 minute. I want to replace all the components that pertain to this such as flame sensor and thermal regulator. Please let me know all the components you recommend replacing because I want to do them all at the same time.

Thanks

Scott