Adding unit tests
Feature vector generation working