Production level ML Code