Add an if statement for initialize in different difficultys