write all featrues for testing
only write needed features
fix classification