Off season training (5)

I’ve been doing well with diet and staying active during the off season and holidays. Trying to swim as much as I can. I just can’t seem to get faster in the water. I have a shrunken lung capacity from cancer-related treatment and am wondering if there’s something I can do to from blowing up in the water so fast. Help!