add difficulty control