fixed image sizing & made difficulty work