Python code for final data mining lab