Added in difficulty UI