Add an argument parser to main class