Added Difficulty Buttons
Added SetState function