Started moving difficulty related functionality into separate class