Changed removeBlock to setBlock so it doesn't destroy the water