Fixing Difficulty.java