Remove assert and add prints