The extended 2nd task on advanced machine learning course