updated comments in classification module