Move main part in a function

pits.py

@@ -85,7 +85,7 @@ def help(args, last_action, conn):
     print('TODO help')
     # TODO help
 
-if __name__ == '__main__':
+def main():
     command = [ 'project', 'task', 'note', 'log', 'info', 'help', 'version', 'init' ]
     handler = [
         project.handle_project,
@@ -120,11 +120,14 @@ if __name__ == '__main__':
         sys.exit(0)
 
     conn = create_connection(DB_FILENAME)
-    last_action = action.read_last_action(conn)
+    last_action = action.read_current(conn)
     logging.debug('Last action: {}'.format(last_action))
     logging.debug(argv)
     logging.debug(argv[2:])
     handler[candidate](argv[2:], last_action, conn)
 
     conn.commit()
-    conn.close()
+    conn.close()
+
+if __name__ == '__main__':
+    main()