added class for hw 3