Minor improvement on zscore rake task