made it take command line arguments