Compact water treatment systems