Demo-day testing + better argument parsing in main