请用面向对象设计思想描述以下类及其相互关系

请用面向对象设计思想描述以下类及其相互关系。

1. 员工类：员工有：性别、姓名、工资、所属部门等字段。
2. 部门类：部门有：名称、部门编号。
3. 测试类：请在测试类中创建属于不同部门的员工，并且将员工存入集合中。再从集合中找出部门总工资中最高的部门。

员工类：

public class Staff extends Department{
    private String name;
    private int salary;
    private String gender;
    private Department department;
	public Staff() {
		super();
	}
	public Staff(String departmentName, String number) {
		super(departmentName, number);
	}
	public Staff(String name, int salary, String gender, Department department) {
		super();
		this.name = name;
		this.salary = salary;
		this.gender = gender;
		this.department = department;
	}
	public String getName() {
		return name;
	}
	public void setName(String name) {
		this.name = name;
	}
	public int getSalary() {
		return salary;
	}
	public void setSalary(int salary) {
		this.salary = salary;
	}
	public String getGender() {
		return gender;
	}
	public void setGender(String gender) {
		this.gender = gender;
	}
	public Department getDepartment() {
		return department;
	}
	public void setDepartment(Department department) {
		this.department = department;
	}
	@Override
	public String toString() {
		return "Staff [name=" + name + ", salary=" + salary + ", gender=" + gender + ", department=" + department + "]";
	}
}

部门类：

public class Department {
	    private String departmentName;
	    private String number;
		public Department() {
			super();
		}
		public Department(String departmentName, String number) {
			super();
			this.departmentName = departmentName;
			this.number = number;
		}
		public String getDepartmentName() {
			return departmentName;
		}
		public void setDepartmentName(String departmentName) {
			this.departmentName = departmentName;
		}
		public String getNumber() {
			return number;
		}
		public void setNumber(String number) {
			this.number = number;
		}
		@Override
		public String toString() {
			return "Department [departmentName=" + departmentName + ", number=" + number + "]";
		}
	}

测试类：

import java.util.ArrayList;

public class Test {
    public static void main(String[] args) {
        Department department1 = new Department("东厂","9527");
        Department department2 = new Department("西厂","1314");
        Staff staff1 =new Staff("王小二",4000,"男",department1);
        Staff staff2 =new Staff("刘小三",4500,"女",department2);
        Staff staff3 =new Staff("二狗",3900,"男",department2);
        Staff staff4 =new Staff("翠花",4450,"女",department1);
        ArrayList<Staff> arrayList = new ArrayList<Staff>();
        arrayList.add(staff1);
        arrayList.add(staff2);
        arrayList.add(staff3);
        arrayList.add(staff4);
        int sum1 =0;
        int sum2 = 0;
        for (int i = 0; i < arrayList.size(); i++) {
        	int salary = arrayList.get(i).getSalary();        	
			if (arrayList.get(i).getDepartment().getDepartmentName() == "东厂") {
				sum1 = sum1 + salary;
			}else if(arrayList.get(i).getDepartment().getDepartmentName() == "西厂"){
				sum2 = sum2 + salary;
			}
		}
        System.out.println("东厂的总工资为：" + sum1);
		System.out.println("西厂的总工资为：" + sum2);
        if(sum1>sum2) {
        	System.out.println("东厂的工资比西厂的高");
        }else if(sum1 < sum2){
        	System.out.println("西厂的工资比东厂的高");
        }else {
        	System.out.println("两个部门的工资一样");
        }
    }
}