[Rake] Improve update task