Python Immersed Boundary Method code