Improving Water Treatment Using Reinforcement Learning