improve edit staff function