Class Solution Python