Interviewbit Machine Learning Interview Question