Feature selection works
Merge pull request #23 from RitwikGupta/master

Feature selection works