added class for layout