Aquariux Interview Question

How can you improve the code?