Python - Insert a new node at the end of the Linked List

Insert a new node at the end of the Linked List.py

def insert_at_end(self, newElement):
  
  #1 & 2 & 3. allocate node, assign data element
  #          assign null to the next of new node
  newNode = Node(newElement)
  
  #4. Check the Linked List is empty or not,
  #   if empty make the new node as head 
  if(self.head == None):
    self.head = newNode
    return
  else:
    
    #5. Else, traverse to the last node
    temp = self.head
    while(temp.next != None):
      temp = temp.next
    
    #6. Change the next of last node to new node  
    temp.next = newNode