Improving Locomotion in Underwater Soft Robots