Accept test size as a command line arg