Added main and handle_connection