better results :). Getting around 80% of accuracy