package mafiaclient;

/**
 * MafiaClient.java
 * Contains the MafiaClient class
 * @author Cory Gehr (cmg5573)
 */
public class MafiaClient {
    
    public static String HOST_NAME;
    public static int PORT_NUMBER = 65004;
    
    /**
     * main()
     * Entry point for the MafiaClient application
     * @param args Command Line arguments
     */
    public static void main(String[] args) {
        
        if(args.length == 2) {
            // Host name set
            HOST_NAME = args[0];
            PORT_NUMBER = Integer.parseInt(args[1]);
        } else {
            // Host name not set, run on localhost
            HOST_NAME = "localhost";
            PORT_NUMBER = 65004;
        }
        
        // Start client connection
        ClientServerConnectionStarter connection = 
                new ClientServerConnectionStarter(HOST_NAME, PORT_NUMBER);
        connection.start();
        
    }
}

package mafiaclient;

/**
 * MafiaClient.java
 * Contains the MafiaClient class
 * @author Cory Gehr (cmg5573)
 */
public class MafiaClient {
    
    public static String HOST_NAME;
    public static int PORT_NUMBER = 65004;
    
    /**
     * main()
     * Entry point for the MafiaClient application
     * @param args Command Line arguments
     */
    public static void main(String[] args) {
        
        if(args.length > 0) {
            // Host name set
            HOST_NAME = args[0];
        } else {
            // Host name not set, run on localhost
            HOST_NAME = "localhost";
        }
        
        // Start client connection
        ClientServerConnectionStarter connection = 
                new ClientServerConnectionStarter(HOST_NAME, PORT_NUMBER);
        connection.start();
        
    }
}

--- 
+++ 
@@ -17,12 +17,14 @@
      */
     public static void main(String[] args) {
         
-        if(args.length > 0) {
+        if(args.length == 2) {
             // Host name set
             HOST_NAME = args[0];
+            PORT_NUMBER = Integer.parseInt(args[1]);
         } else {
             // Host name not set, run on localhost
             HOST_NAME = "localhost";
+            PORT_NUMBER = 65004;
         }
         
         // Start client connection
Use CLI for arguments in client
