Jail 'greatly improved' despite staff shortages