developed python and couette flow code