added my own class