python for challenge of pyrax -- employee code challenge