Program now takes command line arguments