PML final project code and writeup