Идеи для доработки:

1) Уменьшить количество поршней и их расположение.
Вода всегда течёт в направлении ямы, которая находится ближе пяти блоков. Приметно так:
Поэтому хватит 2 поршней на одну сторону (экономим 48 поршней, рэда и рэдфакелов).

2)Кроме того (ну это кому как нравится), можно иначе расположить и подключить нижние поршни, чтобы сделать пол ровным. Но тогда лампы в таком случае обязательны.

3)Небольшая конструкция для кнопки "нажал и забыл". После нажатия сигнал сохраняется ~9-10 секунд. Хорошо влезла в пустое пространство в полу (поршень в схеме - липкий):