diff --git a/CodeGraph/CodeGraph.cpp b/CodeGraph/CodeGraph.cpp
index f7a4181..39a77a1 100644
--- a/CodeGraph/CodeGraph.cpp
+++ b/CodeGraph/CodeGraph.cpp
@@ -37,17 +37,15 @@ using namespace SVF;
 using namespace llvm;
 using namespace std;
 
-static llvm::cl::opt<std::string> InputFilename(cl::Positional,
-        llvm::cl::desc("<input bitcode>"), llvm::cl::init("-"));
 
 int main(int argc, char ** argv) {
 
     int arg_num = 0;
     char **arg_value = new char*[argc];
     std::vector<std::string> moduleNameVec;
-    LLVMUtil::processArguments(argc, argv, arg_num, arg_value, moduleNameVec);
-    cl::ParseCommandLineOptions(arg_num, arg_value,
-                                "Whole Program Points-to Analysis\n");
+    moduleNameVec = OptionBase::parseOptions(
+            arg_num, arg_value, "Teaching-Software-Analysis Assignment 4", "[options] <input-bitcode...>"
+    );
 
     SVFModule* svfModule = LLVMModuleSet::getLLVMModuleSet()->buildSVFModule(moduleNameVec);