Debug a failing ML classifier | OpenAI Interview Question