added code from class