Fix rake task and add a badge