fixing bug with feature generation, working on learning