Add rake task and helper class for adding states