diff --git a/joker_test.py b/joker_test.py
index 34bfd6f..7e6337c 100644
--- a/joker_test.py
+++ b/joker_test.py
@@ -2,25 +2,25 @@ import argparse
 import sys
 from joker import Joker
 
-def arg_parse(): 
-    p = argparse.ArgumentParser(description = 'Joker cli interface') 
-    p.add_argument('-i', '--identity', help = 'Path to identity file', default = None) 
-    p.add_argument('-H', '--host', help = 'destination host') 
-    p.add_argument('-p', '--port', help = 'destination port', default = 22 ) 
-    p.add_argument('-u', '--user', help = 'username', default = "root" ) 
-    p.add_argument('-P', '--password', help = 'username', default = None ) 
-    return p.parse_args() 
+def arg_parse():
+    p = argparse.ArgumentParser(description = 'Joker cli interface')
+    p.add_argument('-i', '--identity', help = 'Path to identity file', default = None)
+    p.add_argument('-H', '--host', help = 'destination host')
+    p.add_argument('-p', '--port', help = 'destination port', default = 22, type = int )
+    p.add_argument('-u', '--user', help = 'username', default = "root" )
+    p.add_argument('-P', '--password', help = 'username', default = None )
+    return p.parse_args()
 
 def main():
-    args = arg_parse() 
+    args = arg_parse()
 
     print args
 
-    j = Joker(args.identity) 
-    j.addNode("EntryPoint", args.host, int(args.port), args.user, args.password) 
+    j = Joker(args.identity)
+    j.addNode("EntryPoint", args.host, args.port, args.user, args.password)
+
+    print j.discover()
 
-    print j.discover() 
-    
 
 if __name__ == '__main__':
     sys.exit(main())