Removes some unnecessary tests
Fixes formatting
Added set difficulty to StateManager