Remove print statements and fix indentation.
Use full data set instead of sample.