code refactored
added staff class