Basic Machine Learning Code Review