Add comments to solution.java