Initial water simulation code and test