Add public modifier to class, which enable run with sbt, but remember to change it when submit to OJ, as well as the class name