Flatten, Then Find Density — hard Machine Learning in Code problem | Incognition