Add gen-class