added class header to model