Add water classification