Water management for agriculture