Added more cases to ML learning