edited out testing code from class