Added annotation to staff model and staff tests