Fix accuracy. Probably should add more test examples.