Simple python wc like script