added logic to mark water