What is your stroke? If you backstroke, freestyle, or butterfly you will want to concentrate on developing upper body power. If you are a breaststroker you'll want to spend considerably more time on the lower body. In any case, I'd suggest adding OLs and other basic strength exercises (especially upper body pulls). High intensity medicine ball work (especially throws like chest passes and overhead pass) would also be great for developing power.