Water treatment for arsenic removal