Add swimming states
Don't apply gravity when swimming