Got accuracy up to 82% on test data