What can I put in my hair so it wont go so frizzy?

I naturally have wavy hair so whenever I get out the shower it gets REALLY frizzy and crazy! I usually straighten it to calm it down. I just found out that Im going to have to take swimming classes for my high school and I dont want to come out of the pool with my hair super crazy! Is there something I can put so It wont frizz.