I need to get better at treading water. Any tips or suggestions?

It's all about flexibility and coordination. Look up some hip flexor and leg stretches and do them religiously. Then watch some YouTube videos on the eggbeater technique and practice on dry land, just laying on your back , then practice in the water while holding the wall… then progress to timed sets in the water.