Adding argument parser
Adding an argument parser to main.go