Add code from class