Depending on difficulty, adding [Streams](https://docs.oracle.com/javase/8/docs/api/java/util/stream/Stream.html) might be helpful.
