updated files
added class back