Final code from class