Fix Swim and balance
Merge pull request #15 from jackstah/master

Fix Swim and balance