Removed the bias and tested on a different dataset, accuracy sucks