Started improving ML output