Java puzzle code