Rewrite rake gem task