difficulty logic added