Added another class to the design