Updated calculate_uv function

Checks only if it is currently a fluid cell and nothing else, maybe we
should change this and check if the other cell is also a fluid cell.