Improve version update rake task