The federal government has flagged new laws to keep the NBN under public ownership.