Code works but looks not quite right


#1

https://www.codecademy.com/en/courses/python-intermediate-en-egNXj/1/4

This is my code and it is working .

class Car(object):
    condition = "new"
    def __init__(self, model, color, mpg):
        self.model = model
        self.color = color
        self.mpg   = mpg
        
    condition="new"
    
    def drive_car(self):
        self.condition="used"
        return self.condition
    
    
    def display_car(self):
        return 'This is a %s %s with %s MPG.' % (self.color, self.model, str(self.mpg))

class ElectricCar(Car):
    def __init__(self,model,color,mpg, battery_type):
        self.model=model
        self.color=color
        self.mpg=mpg
        self.battery_type=battery_type
        
    
    def drive_car(self):
        self.condition="like new"
        return self.condition


my_car = ElectricCar("DeLorean", "silver", 88,"molten salt")


print my_car.condition
my_car.drive_car()
print my_car.condition

10. Problem
#2

You do not need this line twice.

This method is a setter and does not need a return value. It may be removed in both instances.