Must add new class in rakefile