Added mains to trained objects for testing