Add graph generation from main function
Add console argument parsing from console_scripts