Added some logic to difficulty settings