Correctly adding class of ship to locations generated.