finish improving tests