third assignment for advanced machine learning 6540459