Aquariux interview question

How can you improve the code?