move class out of rake task