Using command line flags for simplicity