Added init in main class, fixed bug where user doesn't specify args.