added class and extended it