added arg to set size and number of ships