add swimming AI
more efficient now