Made the ML part able to handle string features